package cn.lg.soar.common.algorithm;


import cn.lg.soar.common.util.data.DataUtil;
import org.bouncycastle.jce.provider.BouncyCastleProvider;

import javax.crypto.BadPaddingException;
import javax.crypto.Cipher;
import javax.crypto.IllegalBlockSizeException;
import javax.crypto.NoSuchPaddingException;
import javax.crypto.spec.SecretKeySpec;
import java.security.*;

/**
 * SM4加密工具（对称加密）
 * @author luguoxiang
 * 开源项目：https://gitee.com/lgx1992/lg-soar 求star！请给我star！请帮我点个star！
 */
public class SM4Utils {

    static {
        Security.addProvider(new BouncyCastleProvider());
    }

    public static final String ALGORITHM_NAME = "SM4";
    // 加密算法/分组加密模式/分组填充方式
    // PKCS5Padding-以8个字节为一组进行分组加密
    // 定义分组加密模式使用：PKCS5Padding
    public static final String ALGORITHM_NAME_ECB_PADDING = "SM4/ECB/PKCS5Padding";
    private Cipher ENCRYPT;
    private Cipher DECRYPT;

    public SM4Utils(byte[] secret){
        this.ENCRYPT = getCipher(Cipher.ENCRYPT_MODE, secret);
        this.DECRYPT = getCipher(Cipher.DECRYPT_MODE, secret);
    }

    public SM4Utils(String secret){
        this(secret.getBytes());
    }

    /**
     * 加密
     * @param data 数据
     * @return
     */
    public byte[] encrypt(byte[] data){
        try {
            return ENCRYPT.doFinal(data);
        } catch (IllegalBlockSizeException | BadPaddingException e) {
            throw new AlgorithmException(e);
        }
    }

    /**
     * 加密（输出base64）
     * @param data 数据
     * @return
     */
    public String encryptToBase64(byte[] data){
        return DataUtil.byteToUrlBase64(encrypt(data));
    }

    /**
     * 解密
     * @param data 数据
     * @return
     */
    public byte[] decrypt(byte[] data){
        try {
            return DECRYPT.doFinal(data);
        } catch (IllegalBlockSizeException | BadPaddingException e) {
            throw new AlgorithmException(e);
        }
    }

    /**
     * 解密（输入base64）
     * @param data 数据（base64）
     * @return
     */
    public byte[] decryptOfBase64(String data){
        return decrypt(DataUtil.urlBase64ToByte(data));
    }

    /**
     * 加密
     * @param secret 密钥
     * @param data 数据
     * @return
     */
    public static byte[] encrypt(byte[] secret, byte[] data) {
        Cipher cipher = getCipher(Cipher.ENCRYPT_MODE, secret);
        try {
            return cipher.doFinal(data);
        } catch (Exception e) {
            throw new AlgorithmException(e);
        }
    }

    /**
     * 加密（输出base64）
     * @param secret 密钥
     * @param data 数据
     * @return
     */
    public static String encryptToBase64(byte[] secret, byte[] data) {
        return DataUtil.byteToUrlBase64(encrypt(secret, data));
    }

    /**
     * 解密
     * @param secret 密钥
     * @param data 数据
     * @return
     */
    public static byte[] decrypt(byte[] secret, byte[] data){
        Cipher cipher = getCipher(Cipher.DECRYPT_MODE, secret);
        try {
            return cipher.doFinal(data);
        } catch (Exception e) {
            throw new AlgorithmException(e);
        }
    }

    /**
     * 解密（输入base64）
     * @param secret 密钥
     * @param data 数据（base64）
     * @return
     */
    public static byte[] decryptOfBase64(byte[] secret, String data){
        return decrypt(secret, DataUtil.urlBase64ToByte(data));
    }

    /**
     * 创建可复用的实例
     * @param secret 密钥
     * @return
     */
    public static SM4Utils create(byte[] secret) {
        return new SM4Utils(secret);
    }

    /**
     * 创建可复用的实例
     * @param secret 密钥
     * @return
     */
    public static SM4Utils create(String secret) {
        return new SM4Utils(secret);
    }


    /**
     * 生成ECB暗号
     *
     * @param mode 模式(加密或解密)
     * @param secret  密钥
     * @return
     * @throws Exception
     * @explain ECB模式（电子密码本模式：Electronic codebook）
     */
    private static Cipher getCipher(int mode, byte[] secret) {
        Key key = new SecretKeySpec(secret, ALGORITHM_NAME);
        try {
            Cipher cipher = Cipher.getInstance(SM4Utils.ALGORITHM_NAME_ECB_PADDING, BouncyCastleProvider.PROVIDER_NAME);
            cipher.init(mode, key);
            return cipher;
        } catch (NoSuchAlgorithmException | NoSuchProviderException | NoSuchPaddingException | InvalidKeyException e) {
            throw new AlgorithmException(e);
        }
    }

}
